/* USER CODE BEGIN Header */
/**
 ******************************************************************************
 * @file           : main.c
 * @brief          : Main program body
 ******************************************************************************
 * @attention
 *
 * Copyright (c) 2025 STMicroelectronics.
 * All rights reserved.
 *
 * This software is licensed under terms that can be found in the LICENSE file
 * in the root directory of this software component.
 * If no LICENSE file comes with this software, it is provided AS-IS.
 *
 ******************************************************************************
 */
/* USER CODE END Header */
/* Includes ------------------------------------------------------------------*/
#include "main.h"
#include "adc.h"
#include "tim.h"
#include "usart.h"
#include "gpio.h"

/* Private includes ----------------------------------------------------------*/
/* USER CODE BEGIN Includes */
#include <stdlib.h>
#include <stdio.h>
#include <string.h>

#include "led.h"
#include "lcd.h"
#include "key.h"
#include "i2c_hal.h"
#include "eeprom.h"
#include "mcp4017.h"
/* USER CODE END Includes */

/* Private typedef -----------------------------------------------------------*/
/* USER CODE BEGIN PTD */

/* USER CODE END PTD */

/* Private define ------------------------------------------------------------*/
/* USER CODE BEGIN PD */

/* USER CODE END PD */

/* Private macro -------------------------------------------------------------*/
/* USER CODE BEGIN PM */

/* USER CODE END PM */

/* Private variables ---------------------------------------------------------*/

/* USER CODE BEGIN PV */
// LED
uint8_t led_buf = 0x00;

// TIM
uint32_t tim_counter = 0;

// PWM
float pwm_duty = 0;
// uint32_t pwm_pulse = 5000;
uint32_t pwm_pulse = 0;
uint32_t pwm_counter = 1000;

// Capture
float freq0;
float freq1;

// ADC
float adc_r37;
float adc_r38;

// EEPROM
uint8_t startup_time;
float adc_mcp4017;

// UART
const char *hello = "Hello World!\r\n";
uint8_t rx_buf[100];
/* USER CODE END PV */

/* Private function prototypes -----------------------------------------------*/
void SystemClock_Config(void);
/* USER CODE BEGIN PFP */

/* USER CODE END PFP */

/* Private user code ---------------------------------------------------------*/
/* USER CODE BEGIN 0 */
float ADC_GetR37(void)
{
  uint16_t adc = 0;

  HAL_ADC_Start(&hadc2);
  HAL_ADC_PollForConversion(&hadc2, 2);
  adc = HAL_ADC_GetValue(&hadc2);
  HAL_ADC_Stop(&hadc2);

  return (float)adc * 3.3f / 4095.0f;
}
float ADC_GetR38(void)
{
  uint16_t adc = 0;
  ADC_ChannelConfTypeDef sConfig = {0};

  sConfig.Channel = ADC_CHANNEL_11;
  sConfig.Rank = ADC_REGULAR_RANK_1;
  sConfig.SamplingTime = ADC_SAMPLETIME_640CYCLES_5;
  sConfig.SingleDiff = ADC_SINGLE_ENDED;
  sConfig.OffsetNumber = ADC_OFFSET_NONE;
  sConfig.Offset = 0;
  if (HAL_ADC_ConfigChannel(&hadc1, &sConfig) != HAL_OK)
  {
    Error_Handler();
  }

  HAL_ADC_Start(&hadc1);
  HAL_ADC_PollForConversion(&hadc1, 2);
  adc = HAL_ADC_GetValue(&hadc1);
  HAL_ADC_Stop(&hadc1);

  return (float)adc * 3.3f / 4095.0f;
}

float ADC_GetMCP4017(void)
{
  uint16_t adc = 0;
  ADC_ChannelConfTypeDef sConfig = {0};

  sConfig.Channel = ADC_CHANNEL_5;
  sConfig.Rank = ADC_REGULAR_RANK_1;
  sConfig.SamplingTime = ADC_SAMPLETIME_640CYCLES_5;
  sConfig.SingleDiff = ADC_SINGLE_ENDED;
  sConfig.OffsetNumber = ADC_OFFSET_NONE;
  sConfig.Offset = 0;
  if (HAL_ADC_ConfigChannel(&hadc1, &sConfig) != HAL_OK)
  {
    Error_Handler();
  }

  HAL_ADC_Start(&hadc1);
  HAL_ADC_PollForConversion(&hadc1, 2);
  adc = HAL_ADC_GetValue(&hadc1);
  HAL_ADC_Stop(&hadc1);

  return (float)adc * 3.3f / 4095.0f;
}

void display()
{
  char buf[30];

  // TIM
  sprintf(buf, "tim_counter=%d      ", tim_counter);
  LCD_DisplayStringLine(Line0, (uint8_t *)buf);

  // PWM
  sprintf(buf, "pwm_duty=%.2f%%      ", pwm_duty * 100.0f);
  LCD_DisplayStringLine(Line1, (uint8_t *)buf);

  // Capture
  sprintf(buf, "freq0=%.2fHz      ", freq0);
  LCD_DisplayStringLine(Line2, (uint8_t *)buf);
  sprintf(buf, "freq1=%.2fHz      ", freq1);
  LCD_DisplayStringLine(Line3, (uint8_t *)buf);

  // ADC
  sprintf(buf, "adc_r37=%.2fV      ", adc_r37);
  LCD_DisplayStringLine(Line4, (uint8_t *)buf);
  sprintf(buf, "adc_r38=%.2fV      ", adc_r38);
  LCD_DisplayStringLine(Line5, (uint8_t *)buf);

  // EEPROM
  sprintf(buf, "startup_time=%d      ", startup_time);
  LCD_DisplayStringLine(Line6, (uint8_t *)buf);
  sprintf(buf, "adc_mcp4017=%.2fV      ", adc_mcp4017);
  LCD_DisplayStringLine(Line7, (uint8_t *)buf);
}

void uart_transmit_info(void)
{
  char buf[200];

  // Print All info
  sprintf(buf, "tim_counter=%d,pwm_duty=%.2f%%,freq0=%.2fHz,freq1=%.2fHz,adc_r37=%.2fV,adc_r38=%.2fV,startup_time=%d,adc_mcp4017=%.2fV\r\n", tim_counter, pwm_duty * 100.0f, freq0, freq1, adc_r37, adc_r38, startup_time, adc_mcp4017);
  HAL_UART_Transmit(&huart1, (uint8_t *)buf, strlen(buf), 1000);
}
/* USER CODE END 0 */

/**
 * @brief  The application entry point.
 * @retval int
 */
int main(void)
{

  /* USER CODE BEGIN 1 */

  /* USER CODE END 1 */

  /* MCU Configuration--------------------------------------------------------*/

  /* Reset of all peripherals, Initializes the Flash interface and the Systick. */
  HAL_Init();

  /* USER CODE BEGIN Init */

  /* USER CODE END Init */

  /* Configure the system clock */
  SystemClock_Config();

  /* USER CODE BEGIN SysInit */

  /* USER CODE END SysInit */

  /* Initialize all configured peripherals */
  MX_GPIO_Init();
  MX_TIM2_Init();
  MX_TIM17_Init();
  MX_TIM3_Init();
  MX_ADC1_Init();
  MX_ADC2_Init();
  MX_USART1_UART_Init();
  MX_TIM16_Init();
  /* USER CODE BEGIN 2 */
  // Reset led
  led(0x00);

  // Init Lcd
  LCD_Init();
  LCD_SetBackColor(Black);
  LCD_SetTextColor(White);
  LCD_Clear(Black);

  // Init TIM
  HAL_TIM_Base_Start_IT(&htim17);
  HAL_TIM_Base_Start_IT(&htim16);

  // Init PWM
  HAL_TIM_PWM_Start(&htim2, TIM_CHANNEL_2);

  // Init Capture
  HAL_TIM_IC_Start_IT(&htim2, TIM_CHANNEL_1);
  HAL_TIM_IC_Start_IT(&htim3, TIM_CHANNEL_1);

  // Init I2C
  I2CInit();
  /* USER CODE END 2 */

  /* Infinite loop */
  /* USER CODE BEGIN WHILE */
  startup_time = EEPROM_ReadByte(0x00);
  startup_time++;
  EEPROM_WriteByte(0x00, startup_time);

  HAL_UART_Transmit(&huart1, (uint8_t *)hello, strlen(hello), 1000);
  HAL_UARTEx_ReceiveToIdle_IT(&huart1, rx_buf, sizeof(rx_buf));
  while (1)
  {
    pwm_pulse = HAL_GetTick() % (pwm_counter);
    __HAL_TIM_SetCompare(&htim2, TIM_CHANNEL_2, pwm_pulse); // Set PWM duty
    pwm_duty = (float)pwm_pulse / (float)pwm_counter;

    adc_r37 = ADC_GetR37();
    adc_r38 = ADC_GetR38();
    adc_mcp4017 = ADC_GetMCP4017();

    display();
    led(led_buf);
    /* USER CODE END WHILE */

    /* USER CODE BEGIN 3 */
  }
  /* USER CODE END 3 */
}

/**
 * @brief System Clock Configuration
 * @retval None
 */
void SystemClock_Config(void)
{
  RCC_OscInitTypeDef RCC_OscInitStruct = {0};
  RCC_ClkInitTypeDef RCC_ClkInitStruct = {0};

  /** Configure the main internal regulator output voltage
   */
  HAL_PWREx_ControlVoltageScaling(PWR_REGULATOR_VOLTAGE_SCALE1);

  /** Initializes the RCC Oscillators according to the specified parameters
   * in the RCC_OscInitTypeDef structure.
   */
  RCC_OscInitStruct.OscillatorType = RCC_OSCILLATORTYPE_HSI;
  RCC_OscInitStruct.HSIState = RCC_HSI_ON;
  RCC_OscInitStruct.HSICalibrationValue = RCC_HSICALIBRATION_DEFAULT;
  RCC_OscInitStruct.PLL.PLLState = RCC_PLL_ON;
  RCC_OscInitStruct.PLL.PLLSource = RCC_PLLSOURCE_HSI;
  RCC_OscInitStruct.PLL.PLLM = RCC_PLLM_DIV1;
  RCC_OscInitStruct.PLL.PLLN = 10;
  RCC_OscInitStruct.PLL.PLLP = RCC_PLLP_DIV2;
  RCC_OscInitStruct.PLL.PLLQ = RCC_PLLQ_DIV2;
  RCC_OscInitStruct.PLL.PLLR = RCC_PLLR_DIV2;
  if (HAL_RCC_OscConfig(&RCC_OscInitStruct) != HAL_OK)
  {
    Error_Handler();
  }

  /** Initializes the CPU, AHB and APB buses clocks
   */
  RCC_ClkInitStruct.ClockType = RCC_CLOCKTYPE_HCLK | RCC_CLOCKTYPE_SYSCLK | RCC_CLOCKTYPE_PCLK1 | RCC_CLOCKTYPE_PCLK2;
  RCC_ClkInitStruct.SYSCLKSource = RCC_SYSCLKSOURCE_PLLCLK;
  RCC_ClkInitStruct.AHBCLKDivider = RCC_SYSCLK_DIV1;
  RCC_ClkInitStruct.APB1CLKDivider = RCC_HCLK_DIV1;
  RCC_ClkInitStruct.APB2CLKDivider = RCC_HCLK_DIV1;

  if (HAL_RCC_ClockConfig(&RCC_ClkInitStruct, FLASH_LATENCY_2) != HAL_OK)
  {
    Error_Handler();
  }
}

/* USER CODE BEGIN 4 */

void HAL_TIM_PeriodElapsedCallback(TIM_HandleTypeDef *htim)
{
  if (htim->Instance == TIM16)
  {
    key_update();
  }
  else if (htim->Instance == TIM17)
  {
    tim_counter++;
    uart_transmit_info();
  }
}

void HAL_UARTEx_RxEventCallback(UART_HandleTypeDef *huart, uint16_t Size)
{
  if (huart->Instance == USART1)
  {
    int buf;
    sscanf((const char *)rx_buf, "%d", &buf);
    MCP4017_Write(buf);
    memset(rx_buf, 0, sizeof(rx_buf));
    HAL_UARTEx_ReceiveToIdle_IT(&huart1, rx_buf, sizeof(rx_buf));
  }
}

void HAL_TIM_IC_CaptureCallback(TIM_HandleTypeDef *htim)
{
  if (htim->Instance == TIM2)
  {
    if (htim->Channel == HAL_TIM_ACTIVE_CHANNEL_1)
    {
      uint32_t ic_val = HAL_TIM_ReadCapturedValue(htim, TIM_CHANNEL_1) + 1;
      freq0 = 1000000.0f / ic_val;
    }
  }
  if (htim->Instance == TIM3)
  {
    if (htim->Channel == HAL_TIM_ACTIVE_CHANNEL_1)
    {
      uint32_t ic_val = HAL_TIM_ReadCapturedValue(htim, TIM_CHANNEL_1) + 1;
      freq1 = 1000000.0f / ic_val;
    }
  }
}

void key_release_callback(uint8_t index)
{
  led_buf ^= 0x01 << index;
}

void key_long_press_callback(uint8_t index)
{
  led_buf ^= 0x10 << index;
  switch (index)
  {
  case 3:
    startup_time = 0;
    EEPROM_WriteByte(0x00, 0x00);
    break;
  default:
    break;
  }
}

/* USER CODE END 4 */

/**
 * @brief  This function is executed in case of error occurrence.
 * @retval None
 */
void Error_Handler(void)
{
  /* USER CODE BEGIN Error_Handler_Debug */
  /* User can add his own implementation to report the HAL error return state */
  __disable_irq();
  while (1)
  {
  }
  /* USER CODE END Error_Handler_Debug */
}

#ifdef USE_FULL_ASSERT
/**
 * @brief  Reports the name of the source file and the source line number
 *         where the assert_param error has occurred.
 * @param  file: pointer to the source file name
 * @param  line: assert_param error line source number
 * @retval None
 */
void assert_failed(uint8_t *file, uint32_t line)
{
  /* USER CODE BEGIN 6 */
  /* User can add his own implementation to report the file name and line number,
     ex: printf("Wrong parameters value: file %s on line %d\r\n", file, line) */
  /* USER CODE END 6 */
}
#endif /* USE_FULL_ASSERT */
